from segment_anything import SamPredictor, sam_model_registry
import cv2
import numpy as np

# 假设 masks 是模型输出的掩码数组（形状为 [N, H, W]）
masks = ...  # 替换为实际掩码数据
image = cv2.imread("your_image.jpg")  # 加载原图
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

# 可视化每个掩码
for mask in masks:
    plt.imshow(image)
    plt.imshow(mask, alpha=0.5)
    plt.axis('off')
    plt.show()